package com.siashan.unipay4j.wx.transfer;

import com.alibaba.fastjson.JSON;
import com.siashan.unipay4j.core.IPayConfig;
import com.siashan.unipay4j.core.Result;
import com.siashan.unipay4j.core.transfer.ITransferQuery;
import com.siashan.unipay4j.core.transfer.TransferQueryModel;
import com.siashan.unipay4j.core.transfer.TransferResp;
import com.siashan.unipay4j.core.transfer.TransferType;
import com.siashan.unipay4j.core.util.EnumUtil;
import com.siashan.unipay4j.wx.AbstractWxPay;
import com.siashan.unipay4j.wx.sdk.WXPayConstants;
import lombok.extern.slf4j.Slf4j;

import java.util.HashMap;
import java.util.Map;

/**
 * 微信企业转账查询
 *
 * 用于商户的付款操作进行结果查询，返回付款操作详细结果。
 *
 * 查询付款API只支持查询30天内的订单，30天之前的订单请登录商户平台查询。
 *
 * 注意事项
 *
 * ◆ 如果查询单号对应的数据不存在，那么数据不存在的原因可能是：（1）付款还在处理中；（2）付款处理失败导致付款订单没有落地。在上述情况下，商户首先需要检查该商户订单号是否确实是自己发起的，如果商户确认是自己发起的，则请商户不要直接当做付款失败处理，请商户隔几分钟再尝试查询（请勿付款和查询并发处理），或者商户可以通过相同的商户订单号再次发起付款。如果商户误把还在付款处理中的订单直接当付款失败处理，商户应当自行承担因此产生的所有损失和责任。
 *
 * ◆ XML具有可扩展性，因此返回参数可能会有新增，而且顺序可能不完全遵循此文档规范，如果在解析回包的时候发生错误，请商户务必不要换单重试，请商户联系客服确认付款情况。如果有新回包字段，会更新到此API文档中。
 *
 * ◆ 因为错误代码字段err_code的值后续可能会增加，所以商户如果遇到回包返回新的错误码，请商户务必不要换单重试，请商户联系客服确认付款情况。如果有新的错误码，会更新到此API文档中。
 *
 * ◆ 错误代码描述字段err_code_des只供人工定位问题时做参考，系统实现时请不要依赖这个字段来做自动化处理。
 *
 * @author  siashan
 * @since  v1.0.1
 **/
@Slf4j
public class WxPayTransferQuery extends AbstractWxPay implements ITransferQuery {
    public WxPayTransferQuery() {
    }

    public WxPayTransferQuery(IPayConfig payConfig) {
        super(payConfig);
    }

    /**
     * 查询企业付款结果
     *
     * @param model  查询参数
     * @return  结果
     */
    @Override
    public Result execQuery(TransferQueryModel model) {
        log.info("WxTransferQuery::=>订单{} 转账查询请求到达", model.getOutTradeNo());
        try {
            // 查询订单状态
            Map<String, String> map = new HashMap<>();
            map.put("partner_trade_no", model.getOutTradeNo());
            String requestUrl = TransferType.TO_ACCOUNT.equals(model.getTransferType()) ? WXPayConstants.MMPAYMKTTRANSFERS_GETTRANSFERINFO_URL_SUFFIX : WXPayConstants.MMPAYMKTTRANSFERS_QUERY_BANK_URL_SUFFIX;
            Map<String, String> r = initWxPay().transferQuery(map,requestUrl,true);
            log.debug("返回报文：" + JSON.toJSONString(r));
            String returnCode = r.get(WXPayConstants.RETURN_CODE);
            if (isReturnSuccess(returnCode)) {
                String resultCode = r.get(WXPayConstants.RESULT_CODE);
                if (isResultSuccess(resultCode)) {
                    return Result.tradeSuccess(wrapperResponse(r));
                } else {
                    String errCode = r.get(WXPayConstants.ERR_CODE);
                    String errCodeDes = r.get(WXPayConstants.ERR_CODE_DES);
                    log.info("WxTransferQuery::=>订单{} 转账查询结果：转账失败,错误代码{},错误原因{}", model.getOutTradeNo(), errCode, errCodeDes);
                    return Result.tradeFail(resultCode,errCode, errCodeDes);
                }
            } else {
                String returnMsg = r.get(WXPayConstants.RETURN_MSG);
                log.info("WxTransferQuery::=>订单{} 转账查询通信失败,错误代码{},错误原因{}", model.getOutTradeNo(), returnCode, returnMsg);
                return Result.returnFail(returnCode, returnMsg);
            }

        } catch (Exception e) {
            log.warn("WxTransferQuery=>订单{} 转账查询发生异常", model.getOutTradeNo(), e);
        }
        return Result.exception();
    }

    private TransferResp wrapperResponse(Map<String, String> r) {
        WxTransferState stateEnum = EnumUtil.getIEnum(WxTransferState.class, r.get(WXPayConstants.STATUS));
        TransferResp response = new TransferResp();
        response.setTransferState(stateEnum.getTransferState());
        response.setOutTradeNo(r.get("partner_trade_no"));
        response.setTradeNo(r.get("payment_no"));
        response.setBody(JSON.toJSONString(r));
        return response;
    }
}
